package com.lz.yz.outing.service;

import com.lz.yz.outing.config.WechatConfig;
import com.lz.yz.outing.entity.MessageFrom;
import com.lz.yz.outing.util.MyX509TrustManager;
import net.sf.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Service;

import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.TrustManager;
import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.ConnectException;
import java.net.URL;
import java.util.List;

/**
 * 微信模版推送工具类
 */
@Service
public class WechatPushService {

    private Logger log = LoggerFactory.getLogger(getClass());

    /**
     * 发送https请求
     *
     * @param requestUrl    请求地址
     * @param requestMethod 请求方式（GET、POST）
     * @param outputStr     提交的数据
     * @return JSONObject(通过JSONObject.get(key)的方式获取json对象的属性值)
     */
    public JSONObject httpsRequest(String requestUrl, String requestMethod, String outputStr) {
        JSONObject jsonObject = null;
        try {
            // 创建SSLContext对象，并使用我们指定的信任管理器初始化
            TrustManager[] tm = {new MyX509TrustManager()};
            SSLContext sslContext = SSLContext.getInstance("SSL", "SunJSSE");
            sslContext.init(null, tm, new java.security.SecureRandom());
            // 从上述SSLContext对象中得到SSLSocketFactory对象
            SSLSocketFactory ssf = sslContext.getSocketFactory();
            URL url = new URL(requestUrl);
            HttpsURLConnection conn = (HttpsURLConnection) url.openConnection();
            conn.setSSLSocketFactory(ssf);

            conn.setDoOutput(true);
            conn.setDoInput(true);
            conn.setUseCaches(false);
            // 设置请求方式（GET/POST）
            conn.setRequestMethod(requestMethod);
            // 当outputStr不为null时向输出流写数据
            if (null != outputStr) {
                OutputStream outputStream = conn.getOutputStream();
                // 注意编码格式
                outputStream.write(outputStr.getBytes("UTF-8"));
                outputStream.close();
            }
            // 从输入流读取返回内容
            InputStream inputStream = conn.getInputStream();
            InputStreamReader inputStreamReader = new InputStreamReader(inputStream, "utf-8");
            BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
            String str = null;
            StringBuffer buffer = new StringBuffer();
            while ((str = bufferedReader.readLine()) != null) {
                buffer.append(str);
            }

            // 释放资源
            bufferedReader.close();
            inputStreamReader.close();
            inputStream.close();
            conn.disconnect();
            jsonObject = JSONObject.fromObject(buffer.toString());
        } catch (ConnectException ce) {
            log.error("连接超时：{}", ce);
        } catch (Exception e) {
            log.error("https请求异常：{}", e);
        }
        return jsonObject;
    }


    /**
     * 预览推送模板
     *
     * @param mf     模板信息
     * @param openid 预览人员
     */
    public String reviewUser(MessageFrom mf, String openid,String url) {
        JSONObject json = new JSONObject();
        json.put("touser", openid);//接收者wxName
        json.put("template_id", WechatConfig.TEST_TEMPLATE_ID);//消息模板
        json.put("url", WechatConfig.TEST_URL);//填写url可查看详情
        JSONObject dd = new JSONObject();
        JSONObject dd2 = new JSONObject();
        dd2.put("value", mf.getFirst());
        dd2.put("color", "#173177");
        dd.put("first", dd2);
        JSONObject cc2 = new JSONObject();
        cc2.put("value", mf.getKeyword1());
        cc2.put("color", "#173177");
        dd.put("keyword1", cc2);
        JSONObject ee2 = new JSONObject();
        ee2.put("value", mf.getKeyword2());
        ee2.put("color", "#173177");
        dd.put("keyword2", ee2);
        JSONObject gg2 = new JSONObject();
        gg2.put("value", mf.getKeyword3());
        gg2.put("color", "#173177");
        dd.put("remark", gg2);
        json.put("data", dd);
        System.out.println(json.toString());
        JSONObject js = this.httpsRequest(url, "POST", json.toString());
        return js.toString();
    }

    /**
     * 全推送模板
     *
     * @param mf         模板信息
     * @param openidList 全关注用户
     */
    public void push(MessageFrom mf, List<String> openidList) {
        for (int i = 0; i < openidList.size(); i++) {
            JSONObject json = new JSONObject();
            json.put("touser", openidList.get(i));//接收者wxName
            json.put("template_id", WechatConfig.TEST_TEMPLATE_ID);//消息模板
            json.put("url", WechatConfig.TEST_URL);//填写url可查看详情
            JSONObject dd = new JSONObject();
            JSONObject dd2 = new JSONObject();
            dd2.put("value", mf.getFirst());
            dd2.put("color", "#173177");
            dd.put("first", dd2);
            JSONObject cc2 = new JSONObject();
            cc2.put("value", mf.getKeyword1());//漏洞等级
            cc2.put("color", "#173177");
            dd.put("keyword1", cc2);
            JSONObject ee2 = new JSONObject();
            ee2.put("value", mf.getKeyword2());//漏洞时间
            ee2.put("color", "#173177");
            dd.put("keyword2", ee2);
            JSONObject gg2 = new JSONObject();
            gg2.put("value", mf.getKeyword3());
            gg2.put("color", "#173177");
            dd.put("remark", gg2);
            json.put("data", dd);
            System.out.println(json.toString());
            JSONObject js = this.httpsRequest(WechatConfig.WECHAT_PUSH_URL, "POST", json.toString());
            System.out.println("js==" + js);
        }
    }


}